import sys, pygame

pygame.init()
yanse = (255,255,255)
dizhi = [40,40,40,40]
chicun = width,height = 400,400
screen = pygame.display.set_mode((chicun))

def draw_graph(func,color,dizhi):
    func(screen,color,dizhi)
    pygame.display.flip()
draw_graph(pygame.draw.rect,yanse,dizhi)
